use "$out/data_for_main_regs_event_study_20231203.dta", clear
drop if control==1

append using "$out/additional_data_for_IV_and_placebo_regs_20240416.dta"
replace treated=1 if d==2
drop if d==1 & control==0

append using "$out/additional_data_for_all_regs_controls_d_distrib_as_treated_20250319.dta"




gen NONEMP = 1 if ABS==1
	recode NONEMP (.=0) if ABS==0
	
set seed 12345

keep if d<=6
*keep if d>=3 & d<=6
*keep if NONEMP !=.

gen month_shock = t-12*int(t/12)
	recode month_shock (0=12)
gen month_return = month_shock+d

*keep if treated == 1 | control == 1

forval i=1/8 {
	local j=`i'+8
	rename mean_fq`i'_w w_q`j'
	replace mean_fq`i'_FEj_imp =. if flag_missingFEj_fq`i'==1
	rename mean_fq`i'_FEj_imp FE_q`j'


}
forval i=1/8 {
	local j=8-`i'
	rename mean_lq`i'_w w_q`j'
	replace mean_lq`i'_FEj_imp =. if flag_missingFEj_lq`i'==1
	rename mean_lq`i'_FEj_imp FE_q`j'
}


* reshape *
capture drop FEj_t4 FEj_t5 FEj_t6 FEj_t7 FEj_t8 FEj_t9 FEj_t10 FEj_t11 FEj_t12 
capture drop lnw_restr_t4 lnw_restr_t5 lnw_restr_t6 lnw_restr_t7 lnw_restr_t8 lnw_restr_t9 lnw_restr_t10 lnw_restr_t11 lnw_restr_t12 

reshape long w_q FE_q, i(anon) j(relative_time)


recode teaor1_h1 (-1=99) (.=100)

cap drop date
gen date=t+(relative_time-8)*3
replace date=t+d+(relative_time-9)*3 if relative_time>=8

rename kor kor_orig

rename ev ev_orig
rename t t_orig
rename date t

gen ev=floor((t-1)/12)+2003
merge 1:1 anon t using "$in/admin3_alap.dta", nogen keep(master match) keepusing(vallazon1 wtip1 wh1 feor1_h2 kor)
replace wh1=. if wh1>40
rename vallazon1 vallazon
merge m:1 vallazon ev using "$out/firmquality.dta", nogen keep(master match)

merge 1:1 anon t using "$in/admin3_passziv.dta", nogen keepusing(passziv_ho)
replace passziv_ho=0 if passziv_ho==.

by anon, sort: gegen full_d=min(d)
drop if full_d==. //drop anons who are never in our baseline sample

rename t date
rename t_orig t

by anon, sort: gegen full_t=min(t)

gen sl_relevant1_=passziv_ho if date>=full_t & date<=full_t+1
by anon, sort: gegen sl_relevant1=total(sl_relevant1_)
drop sl_relevant1_

drop if d==. //back to baseline+falsification sample
drop if d>6

keep if control==1 & ev_orig>=2012 & letszam~=. & FE_q~=. //controls only!!


keep vallazon //firm ID at time t (i.e., at the shock)
rename vallazon vallazon_base
gen control_base=1
duplicates drop

merge 1:m vallazon_base using "$out/coworker_regfile_with_ncoworkers_AB_ModTreated_clean.dta", keepusing(control match_sample ev FE_q* relative_time letszam stay* anon) 

gen EMP = 1 if FE_q_w!=.
	recode EMP (.=0) if FE_q_w==.
egen totEMP = total(EMP), by(anon)

gen stay2 = stay if EMP==1

foreach var of varlist EMP stay stay2 {
	gen pre`var' = `var' if relative_time<=7
	egen tot_pre`var' = total(pre`var'), by(anon)
}

keep if (control==1 & ev>=2012 & FE_q~=. & letszam~=. & match_sample==1  & tot_preEMP==8 & tot_prestay2>=4 & tot_prestay2~=.) | control_base==1
keep vallazon_base _merge
duplicates drop

ta _merge

keep if _merge==2
gen firms_exclude=1
keep vallazon_base firms_exclude

save "$out/coworkers_firms_exclude_AB.dta", replace
